import torch
print("CUDA可用:", torch.cuda.is_available())
print("CUDA版本:", torch.version.cuda)
print("GPU设备:", torch.cuda.get_device_name(0) if torch.cuda.is_available() else "None")